#pragma once

#include "Shape.h"

class ShapeFactory
{
public:
    Shape *getShape(std::string shape)
    {
        if (shape.compare("RECTANGLE") == 0) {
            return new Rectangle();
        } else if (shape.compare("SQUARE") == 0) {
            return new Square();
        } else if (shape.compare("CIRCLE") == 0) {
            return new Circle();
        }
        return nullptr;
    }
};